from modelscope import snapshot_download 


# 下载modelscope中的模型 (cache_dir为保存模型的地址，必须统一)
# RAG的Embedding模型
embed_model_dir = snapshot_download('AI-ModelScope/all-mpnet-base-v2', cache_dir="models/models")
# 医学问诊意图识别模型
intent_model_dir = snapshot_download('stevenzhun/RoBERTa-Chinese-Med-Inquiry-Intention-Recognition-base', cache_dir='models/models')

# 打印地址
print(embed_model_dir)
print(intent_model_dir)




